package com.bootdo.clouddoES.config;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * @title: RabbitConfig
 * @Author gjt
 * @Date: 2020-12-11
 * @Description：消息确认回调函数
 */
/**
 * ①消息推送到server，但是在server里找不到交换机
 * 结论： ①这种情况触发的是 ConfirmCallback 回调函数。
 * ②消息推送到server，找到交换机了，但是没找到队列
 * 结论：②这种情况触发的是 ConfirmCallback和RetrunCallback两个回调函数。
 * ③消息推送到sever，交换机和队列啥都没找到
 * 结论： ③这种情况触发的是 ConfirmCallback 回调函数。
 * ④消息推送成功
 * 结论： ④这种情况触发的是 ConfirmCallback 回调函数。
 */
@Configuration
public class RabbitConfig {

    @Bean
    public RabbitTemplate createRabbitTemplate(ConnectionFactory connectionFactory) {
        Logger log = LoggerFactory.getLogger(RabbitTemplate.class);

        RabbitTemplate rabbitTemplate = new RabbitTemplate();
        rabbitTemplate.setConnectionFactory(connectionFactory);

        // 消息发送失败返回到队列中, yml需要配置 publisher-returns: true

        //设置开启Mandatory,才能触发回调函数,无论消息推送结果怎么样都强制调用回调函数
        rabbitTemplate.setMandatory(true);

        // 消息确认, yml需要配置 publisher-confirms: true
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                System.out.println("ConfirmCallback:     " + "相关数据：" + correlationData);
                System.out.println("ConfirmCallback:     " + "确认情况：" + ack);
                System.out.println("ConfirmCallback:     " + "原因：" + cause);

                if (ack) {
                    // log.debug("消息发送到exchange成功,id: {}", correlationData.getId());
                } else {
                    log.info("消息发送到exchange失败,原因: {}", cause);
                }
            }
        });

        // 消息返回, yml需要配置 publisher-returns: true
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                System.out.println("ReturnCallback:     " + "消息：" + message);
                System.out.println("ReturnCallback:     " + "回应码：" + replyCode);
                System.out.println("ReturnCallback:     " + "回应信息：" + replyText);
                System.out.println("ReturnCallback:     " + "交换机：" + exchange);
                System.out.println("ReturnCallback:     " + "路由键：" + routingKey);

                log.info("消息：{} 发送失败, 应答码：{} 回应信息：{} 交换机: {}  路由键: {}", message, replyCode, replyText, exchange, routingKey);
            }
        });

        return rabbitTemplate;
    }

}
